package j2024.j202412;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class j1206 {
    /**
     * 15. 三数之和
     *
     * 给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j],
     * nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。
     * 请你返回所有和为 0 且不重复的三元组。
     *
     * 注意：答案中不可以包含重复的三元组。
     * @param nums
     * @return
     */
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        int n = nums.length;
        List<List<Integer>> ret = new ArrayList<>();
        for (int i = 0; i < n;) {
            if(nums[i]>0){
                break;
            }
            int tmp = -nums[i];
            int left = i+1,right = n-1;
            while(left<right){
                if(nums[left]+nums[right]>tmp){
                    right--;
                }else if(nums[left]+nums[right]<tmp){
                    left++;
                }else {
                    ret.add(new ArrayList<>(Arrays.asList(nums[i],nums[left],nums[right])));
                    left++;right--;
                    while(left<right && nums[left]==nums[left-1]){
                        left++;
                    }
                    while(left<right && nums[right]==nums[right+1]){
                        right--;
                    }
                }
            }
            i++;
            while(i<n && nums[i]==nums[i-1]){
                i++;
            }
        }
        return ret;
    }
}
